building a model in python